solving optimization problems in python